// 定义一个中间件函数，用于添加res.success等方法
function addResMethod() {
  return function (req, res, next) {
    // 给res对象添加一个success方法
    res.success = function (data = {}, message = '请求成功') {
      // 参数个数获取
      if (arguments.length === 1 && typeof data === 'string') {
        message = data
        res.send({
          code: 200,
          message: message,
        })
        return
      }
      res.send({
        code: 200,
        message: message,
        data: data,
      })
    }
    res.error = function (code = 200, message = '请求失败') {
      res.status(code).send({
        code: code,
        message: message,
      })
    }
    next() // 继续执行路由处理
  }
}

module.exports = addResMethod
